#!/usr/bin/python3
# -*- coding:utf-8 -*-
""""
   @Author: 叶建伟
   @Email: yejw46507@hundsun.com
   @Time: 2022/6/8 9:37
   @File: 核查场景新增_客户证件逾期.py
"""
import time

from 基础公共脚本.步骤检查_hui2 import check_table, check_tips_msg
from 基础公共脚本.公共方法_hui2 import open_menu,FE
from 基础公共脚本.common_util import auto_fill_form,click_submit,check_label
from 数据核查.testUnit.check_scene_util import scene_check_index,scene_input_index

def check_scene_add_overdue(input_data=None, ui_expect=None):
    open_menu({'子系统名称': '反洗钱系统', '菜单名称': '数据核查/核查场景'})
    FE.button("新增场景").click()
    time.sleep(1)
    add_overdue(input_data)
    click_submit()
    # 检查查询结果
    check_table(input_data=None, ui_expect=ui_expect)

def check_scene_add_overdue_contrary(input_data=None, ui_expect=None):
    open_menu({'子系统名称': '反洗钱系统', '菜单名称': '数据核查/核查场景'})
    FE.button("新增场景").click()
    time.sleep(1)
    add_overdue(input_data)
    click_submit()
    # 检查查询结果
    check_tips_msg(ui_expect=ui_expect)

def add_overdue(input_data):
    auto_fill_form("场景类型", input_data.get('必填项'))
    time.sleep(1)
    if input_data.get("非必填项"):
        temp = input_data.get("非必填项")
        if temp.get("证件即将过期"):
            check_label("证件即将过期")
            tempList = temp.get("证件即将过期").split("&")
            scene_input_index("证件即将过期", "即将过期天数", tempList[0], 2)
            if tempList[1]:
                scene_check_index("证件即将过期", tempList[1], 3)
            if tempList[2] == "受益所有人证件":
                pass
            else:
                if tempList[2]:
                    scene_check_index("证件即将过期", tempList[2], 4)
            if tempList[3]:
                scene_check_index("证件即将过期", tempList[3], 5)
            scene_input_index("证件即将过期", "即将过期天数", tempList[4], 6)
        if temp.get("证件过期时间"):
            check_label("证件过期时间")
            tempList = temp.get("证件过期时间").split("&")
            scene_input_index("证件过期时间", "过期天数", tempList[0], 2)
            if tempList[1]:
                scene_check_index("证件过期时间", tempList[1], 3)
            if tempList[2] == "受益所有人证件":
                pass
            else:
                if tempList[2]:
                    scene_check_index("证件过期时间", tempList[2], 4)
            if tempList[3]:
                scene_check_index("证件过期时间", tempList[3], 5)
            scene_input_index("证件过期时间", "过期天数", tempList[4], 6)
